package com.example.demo.saToken.controller;

import cn.dev33.satoken.stp.StpUtil;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.core.env.Environment;
import org.springframework.stereotype.Controller;
import org.springframework.ui.Model;
import org.springframework.web.bind.annotation.RequestMapping;

import jakarta.annotation.Resource;
import java.util.Arrays;

/**
 * @ClassName IndexController
 * @Description 入口控制器
 * @Author MingChang.Wei
 * @Date 2024/9/20
 **/
@Controller
public class IndexController {
    
    @Value("${frontend.url:http://localhost:3000}")
    private String frontendUrl;
    
    @Resource
    private Environment environment;
    
    @RequestMapping({"/", "/index"})
    public String index(Model model) {
        model.addAttribute("message", "前后端联合交互的验证码插件");
        
        // 检查用户是否已登录
        if (StpUtil.isLogin()) {
            // 安全的Session方式：Sa-Token会自动在Cookie中设置token，前端通过API获取用户信息
            return "redirect:" + frontendUrl;
        } else {
            // 未登录情况的处理
            String[] activeProfiles = environment.getActiveProfiles();
            boolean isDev = Arrays.asList(activeProfiles).contains("dev") || 
                           Arrays.asList(activeProfiles).contains("local");
            
            if (isDev) {
                // 开发环境：显示友好的重定向页面
                model.addAttribute("frontendUrl", frontendUrl);
                return "dev-redirect";
            } else {
                // 生产环境：直接重定向到登录页面
                return "redirect:/sa-token/login.html";
            }
        }
    }
}
